#pragma once

#include "TheWorld.h"

// TODO: refactor this class

enum RadarViewMode
{
    RADARMODE_ALL,
    RADARMODE_WITHOUT_HOSTILE
};


class Radar
{
public:
    Radar();
    void setRadius(int radius);
    void setViewMode(RadarViewMode mode);
    void setWorld(TheWorld * theWorld);

    std::vector<IRobot *> & getRobots();

private:
    RadarViewMode _mode;
    int _radius;

    TheWorld * _world;
};
